#pragma once

#include <iostream>
#include <cstring>
#include <string>
#include <cerrno>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/wait.h>
#include <signal.h>
#include <pthread.h>
#include <functional>

#include "Log.hpp"
#include "Common.hpp"
#include "IntAddr.hpp"
#include "ThreadPool.hpp"

#define BACKLOG 8

using namespace LogModule;
using namespace ThreadPoolModule;

static const uint16_t gport = 8082;
//处理的任务类型, 返回值为string , 参数为string
//using handler_t = std::function<std::string(std::string)>;
using handler_t = std::function<std::string(std::string&)>;

class TcpServer
{
    using task_t = std::function<void()>;
    struct ThreadData
    {
        int sockfd;
        TcpServer *self;
    };
public:
    TcpServer(handler_t handler, int port = gport)
        : _handler(handler)
        , _port(port)
        , _isrunning(false)
        {}
    //初始化服务器
    void InitServer()
    {
        //1. 创建tcp socket
        //tcp所创建的socket只用来起到监听作用
        _listensockfd = ::socket(AF_INET, SOCK_STREAM, 0);
        if(_listensockfd < 0)
        {
            LOG(LogLevel::FATAL) << "socket error";
            Die(SOCKET_ERR);
        }
        LOG(LogLevel::INFO) << "sicket create success, sockfd is: " << _listensockfd;

        //设置选项
        int opt = 1;
        setsockopt(_listensockfd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt));

        //设置网络信息
        struct sockaddr_in local;
        memset(&local, 0, sizeof(local));
        local.sin_family = AF_INET;
        local.sin_port = htons(gport);
        local.sin_addr.s_addr = INADDR_ANY;

        //2.绑定
        int n = ::bind(_listensockfd, CONV(&local), sizeof(local));
        if(n < 0)
        {
            LOG(LogLevel::FATAL) << "bind error";
            Die(BIND_ERR);
        }
        LOG(LogLevel::INFO) << "bind success : " << _listensockfd;

        //3.cs, tcp是面向连接的, 就要求tcp随时随地等待被连接
        //tcp需要把socket设置为监听状态
        n = ::listen(_listensockfd, BACKLOG);
        if(n < 0)
        {
            LOG(LogLevel::FATAL) << "lesson error";
            Die(LISTEN_ERR);
        }
        LOG(LogLevel::INFO) << "listen success, lesonsockfd is: " << _listensockfd;


    }

    void HandlerRequst(int sockfd)
    {
        LOG(LogLevel::INFO) << "HandlerRequest, sockfd is: " << sockfd;
        char inbuffer[4096];

        std::string package; //需要接收完整的报文

        //长任务
        while(true)
        {
            //约定: 用户发来的是一个完整的命令string
            //read读取的是不完整的
            ssize_t n = ::recv(sockfd, inbuffer, sizeof(inbuffer) - 1, 0);
            if(n > 0)
            {
                inbuffer[n] = 0;
                LOG(LogLevel::INFO) << inbuffer;
            
                // std::string echo_str = "server echo# ";
                // echo_str += inbuffer;

                package += inbuffer; //TCP是字节传输, 传输的信息可能是不完整的
                
                //检查是否是完整的报文信息, 并且读取完整的报文,并进行处理
                std::string cmd_result = _handler(package);
                if(cmd_result.empty()) continue; //TODO
                
                //::send(sockfd, echo_str.c_str(), echo_str.size(), 0);
                ::send(sockfd, cmd_result.c_str(), cmd_result.size(), 0); //写入可能也是不完善的, 需要在客户端进行处理
            }
            else if(n == 0)
            {
                //read如果返回值为0, 表示client退出
                LOG(LogLevel::INFO) << "client quit: " << sockfd;
                break;   
            }
            else
            {
                break; //读取失败
            }
        }
        ::close(sockfd); //防止fd泄露
    }

    //类成员方法默认带有this指针
    static void *ThreadEntry(void *args)
    {
        //让线程进行分离, 要不然还是得回收
        pthread_detach(pthread_self());
        ThreadData *data = (ThreadData *)args;
        data->self->HandlerRequst(data->sockfd);
        return nullptr;
    }

    void Start()
    {
        _isrunning = true;
        while(_isrunning)
        {
            //不能直接读取数据
            //1.获取链接 accept
            struct sockaddr_in peer;
            socklen_t perrlen = sizeof(peer); 
            LOG(LogLevel::DEBUG) << "accept ing...";
            int sockfd = ::accept(_listensockfd, CONV(&peer), &perrlen);
            //此时的sockfd才是我们通信的
            if(sockfd < 0)
            {
                LOG(LogLevel::WARNING) << "accept error: " << strerror(errno);
                continue; 
            }
            //获取连接成功了
            LOG(LogLevel::INFO) << "accept success, sockfd is : " << sockfd;
            InetAddr addr(peer);
            LOG(LogLevel::INFO) << "client info: " << addr.Addr();

            //version-0
            //HandlerRequst(sockfd);
            //但是当前版本只能进行单个客户端进行通信, 因为代码执行到recv那里会进行
            //阻塞,等待用户输入, 就无法进入下一次循环, 执行accept来获取新的客户端
            //version-1 多进程版本, 让子进程去执行任务, 父进程不断获取客户端
            //关闭不需要的文件流, 子进程拷贝一份父进程的文件描述符

            // pid_t id = fork();
            // if(id == 0)
            // {
            //     //child
            //     //问题1:子进程继承父进程的文件描述符, 两张, 父子各一张
            //     //1.关闭不需要的fd
            //     ::close(_listensockfd);
                
            //     //子进程需要父进程等待回收, 还是需要等待子进程, 孙子进程
            //     if(fork() > 0) exit(0);//子进程退出
            //     //孙子进程->孤儿进程->1
            //     HandlerRequst(sockfd);
            //     exit(0);
            // }
            // ::close(sockfd); //父进程不需要这个文件描述符, 所以后续的客户端文件
            //描述符都是一样的, 子进程退出会自己回收文件描述符
            //此时不会阻塞, 不需要回收
            // int rid = ::waitpid(id, nullptr, 0);
            // if(rid < 0)
            // {
            //     LOG(LogLevel::WARNING) << "waitpid error";
            // }

            //version-2: 多线程版本, 创建进程所耗费的资源比线程大, 创建线程
            //主线程和新线程是如何看待:文件描述符表, 共享一张文件描述符表!!
            // pthread_t tid;
            // ThreadData *data = new ThreadData;
            // data->sockfd = sockfd;
            // data->self = this;
            // pthread_create(&tid, nullptr, ThreadEntry, data);

            //version-3: 线程池版本, 比较适合处理短任务, 或者是用户量少的情况
            task_t f = std::bind(&TcpServer::HandlerRequst, this, sockfd);
            ThreadPool<task_t>::getInstance()->Equeue([this, sockfd]() { //&this this的声明周期有限, 相当于函数的形参, 不能传引用
                this->HandlerRequst(sockfd);});
        }

    }
    void Stop()
    {
        _isrunning = false;
    }
    ~TcpServer()
    {
    }
private:
    int _listensockfd; //监听
    uint16_t _port;
    bool _isrunning;

    //处理上层任务的入口
    handler_t _handler;
};